🔗 문제 링크

두 수의 합

✔️ 소요된 시간

1H

✨ 수도 코드

오랜만에 돌아오는 김에 쉬운 문제로 찾아왔습니다.

✔️ 접근법

문제를 요약하자면 수열이 주어졌을 때 X를 구할 수 있는 두 수의 합을 구하는 것 입니다.
모든 경우의 수를 고려하는 것은 시간이 너무 오래걸리기에 패스하고 다른 방법을 찾아봐야겠네요.
이 때 든 생각은 바로 이렇습니다.
‼️ 두 수의 합이 X가 되기 위해서 한 숫자가 A라면 다른 숫자는 X-A이다.
좀 더 직관적으로 설명을 해보자면, X/2보다 큰 숫자 하나, 작은 숫자 하나의 합에서만 X가 나올 수 있다는 것입니다.

♦️ 간단한 예시를 들어보자면,
1 2 3 4 5 에서 6을 만들 수 있는 두 수의 합은 (1,5), (2,4) 입니다. 이 때 6/2 = 3을 기준으로 큰 수 하나, 작은 수 하나의 합으로 만들 수 있었네요.

그렇다면 이를 구현하기 위해서 우리는 정렬된 수열이 필요하다는 것을 알 수 있습니다.

✔️주어진 예시

주어진 예시를 보면 다음과 같습니다.
아래와 같은 수열이 문제에서 주어져있었고

이를 정렬하게 되면 다음과 같이 수열이 바뀌게 됩니다.

이 때 우리는 13/2 = 7.5를 기준으로 큰 수 하나, 작은 수 하나의 합으로 13을 만들 수 있다는 것을 볼 수 있습니다.

이를 구현하기 위하여, 정렬된 수열의 양 끝에서 포인터들을 움직이며, 해당 포인터들이 가리키는 숫자의 합이 X가 되는지 확인하는 것이 제가 생각한 코드입니다.

📝 SUDO CODE

1️⃣ 입력을 받아온다
2️⃣ 수열을 정렬한다. 
3️⃣ 수열의 양 끝에 포인터를 두고, 아래의 조건에 맞게 움직이며 반복문을 수행한다. 
    3️⃣-1 앞쪽에서 시작한 포인터가, 뒤쪽에서 시작한 포인터보다 뒤에 있게 된다면 반복문을 종료한다. 
    3️⃣-2 만약 숫자의 합이 목표값보다 크다면 뒤쪽의 포인터를 한칸 앞으로 당긴다. 
    3️⃣-3 만약 숫자의 합이 목표값보다 작다면 앞쪽의 포인터를 한칸 뒤로 민다. 
    3️⃣-4 만약 숫자의 합이 목표값과 같다면 cnt에 값을 1 증가시키고, 포인터를 이동시킨다. 
4️⃣ cnt를 출력한다. 

🖥️ 코드

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;
int main(){
    int n, goal;
    int a, b, cnt = 0;;
    cin >> n;
    a = 0;
    b = n-1;
    vector<int> vec(n,0);
    for(int i = 0; i<n; i++){
        cin >> vec[i];
    }
    sort(vec.begin(), vec.end());
    cin >> goal;
    while(true){
        if(b<=a){
            cout << cnt;
            break;
        }
        if(vec[a]+vec[b] == goal){
            cnt++;
            a+=1;
            b-=1;
        }
        else if(vec[a]+vec[b] > goal){
            b-=1;
        }
        else{
            a+=1;
        }
    }
    return 0;
}

📚 새롭게 알게된 내용

✔️ 투포인터를 사용하여 풀이할 수 있는 간단한 문제였습니다.